# -*- coding: UTF-8 -*-
# *******************************************************************
# File Name: yolo
# > Author: 04000387
# > Created Time: 2024/12/24 10:53
# *******************************************************************
from torch import nn, optim
from torch.utils.data import DataLoader
from torchvision.ops import box_iou, nms


class YoloModel(nn.Module):
    def __init__(self, backbone="resnet18"):
        super().__init__()
